import os
import shutil

# 源目录
source_dir = "../yolo/val/labels"
# 目标目录
target_dir = "../test/text/val/labels"

# 确保目标目录存在
os.makedirs(target_dir, exist_ok=True)

# 遍历源目录下的所有文件
for filename in os.listdir(source_dir):
    if filename.endswith('.txt'):
        # 拆分文件名和扩展名
        base_name, file_extension = os.path.splitext(filename)
        # 生成新的文件名，添加 _cover_text 后缀
        new_base_name = f"{base_name}_cover_text"
        new_filename = f"{new_base_name}{file_extension}"

        # 源文件的完整路径
        source_path = os.path.join(source_dir, filename)
        # 目标文件的完整路径
        target_path = os.path.join(target_dir, new_filename)

        # 复制文件
        shutil.copy2(source_path, target_path)

print("文件复制和重命名完成。")